package leetcode.Two_hundred_to_Three_hundred;

import java.util.ArrayList;
import java.util.List;

/**
 * 257. 二叉树的所有路径
 */
public class Test257 {

    static class TreeNode {
        int val;
        TreeNode left;
        TreeNode right;

        TreeNode(int x) {
            val = x;
        }
    }

//    public List<String> binaryTreePaths(TreeNode root) {
//        List<String> ans = new ArrayList<>();
//        if (root == null) {
//            return ans;
//        }
//        if (root.left == null && root.right == null) {
//            ans.add(root.val + "");
//            return ans;
//        }
//        List<String> left = binaryTreePaths(root.left);
//        for (int i = 0; i < left.size(); i++) {
//            ans.add(String.valueOf(root.val) + "->" + left.get(i));
//        }
//
//        List<String> right = binaryTreePaths(root.right);
//        for (int i = 0; i < right.size(); i++) {
//            ans.add(String.valueOf(root.val) + "->" + right.get(i));
//        }
//
//        return ans;
//    }


    List<String> res = new ArrayList<>();

    public List<String> binaryTreePaths(TreeNode root) {
        if (root == null) {
            return res;
        } else {
            dfs(root, "");
            return res;
        }
    }

    public void dfs(TreeNode root, String temp) {
        int val = root.val;
        temp = temp.equals("") ? String.valueOf(val) : temp + "->" + String.valueOf(val);
        if (root.left != null) {
            dfs(root.left, temp);
        }
        if (root.right != null) {
            dfs(root.right, temp);
        }
        // 直到遍历到叶子节点后，才往 res 里面添加
        if (root.left == null && root.right == null) {
            res.add(temp);
        }
    }
}
